package com.lingang.framework.mybatis;

import cn.hutool.core.date.DateUtil;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.lingang.common.utils.SecurityUtils;
import com.lingang.framework.web.domain.BaseModel;
import java.time.LocalDateTime;
import java.util.Objects;
import org.apache.ibatis.reflection.MetaObject;

/**
 * @author hj
 * @date 2025年05月31日 23:44
 */
public class DefaultDBFieldHandler implements MetaObjectHandler {
    @Override
    public void insertFill(MetaObject metaObject) {
        if (Objects.nonNull(metaObject) && metaObject.getOriginalObject() instanceof BaseModel) {

            BaseModel  baseDO = (BaseModel) metaObject.getOriginalObject();
            // 创建时间为空，则以当前时间为插入时间
            if (Objects.isNull(baseDO.getCreateTime())) {
                baseDO.setCreateTime(DateUtil.date());
            }
            // 更新时间为空，则以当前时间为更新时间
            if (Objects.isNull(baseDO.getUpdateTime())) {
                baseDO.setUpdateTime(DateUtil.date());
            }

            Long userId = SecurityUtils.getUserId();
            // 当前登录用户不为空，创建人为空，则当前登录用户为创建人
            if (Objects.nonNull(userId) && Objects.isNull(baseDO.getCreateBy())) {
                baseDO.setCreateBy(SecurityUtils.getUsername());
            }
            // 当前登录用户不为空，更新人为空，则当前登录用户为更新人
            if (Objects.nonNull(userId) && Objects.isNull(baseDO.getUpdateBy())) {
                baseDO.setUpdateBy(SecurityUtils.getUsername());
            }
        }
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        BaseModel  baseDO = (BaseModel) metaObject.getOriginalObject();
        // 更新时间为空，则以当前时间为更新时间
        Object modifyTime = getFieldValByName("update_time", metaObject);
        if (Objects.isNull(modifyTime)) {
            setFieldValByName("update_time", LocalDateTime.now(), metaObject);
        }

        // 当前登录用户不为空，更新人为空，则当前登录用户为更新人
        Object modifier = getFieldValByName("update_by", metaObject);
        Long userId = SecurityUtils.getUserId() ;
        if (Objects.nonNull(userId) && Objects.isNull(modifier)) {
            setFieldValByName("update_by", SecurityUtils.getUsername(), metaObject);
        }
    }
}
